package at.ac.tuwien.isis.infret.testing;

import static org.junit.Assert.*;

import java.io.File;
import java.util.Iterator;

import org.junit.After;
import org.junit.Assert;
import org.junit.Before;
import org.junit.Test;

import at.ac.tuwien.isis.infret.CorpusReader;

public class CorpusReaderTest {
	
	CorpusReader test;
	
	@Before
	public void setUp() throws Exception {
		//System.out.println(File.separatorChar+"testing");
		test = new CorpusReader("testing"+File.separatorChar);
	}

	@Test
	public void testGetClassIter() {
		Iterator<String> classes = test.getClassIter();
		//Assert.assertEquals(".svn", classes.next());
		Assert.assertEquals("alt.atheism", classes.next());
		Assert.assertEquals("comp.graphics", classes.next());
	}

/*	@Test
	public void testGetDocIter() {
		Iterator<String> docs = test.getDocIter();
		Assert.assertEquals("short testing doc", docs.next());
		Assert.assertEquals("another short testing doc", docs.next());
		Assert.assertEquals("short testing doc 2nd class", docs.next());
		
	}*/

}
